package com.luo.ai.assistant;

import org.dromara.x.file.storage.spring.EnableFileStorage;
import org.springframework.ai.model.ollama.autoconfigure.OllamaChatAutoConfiguration;
import org.springframework.ai.model.ollama.autoconfigure.OllamaEmbeddingAutoConfiguration;
import org.springframework.ai.model.openai.autoconfigure.OpenAiChatAutoConfiguration;
import org.springframework.ai.vectorstore.cassandra.autoconfigure.CassandraVectorStoreAutoConfiguration;
import org.springframework.ai.vectorstore.redis.autoconfigure.RedisVectorStoreAutoConfiguration;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

/**
 * My AI Assistant Application
 *
 * @author luohq
 * @date 2025-01-02
 */
@EnableFileStorage
@SpringBootApplication(
        exclude = {
                //排查OpenAI自动配置
                //OpenAiChatAutoConfiguration.class,
                //排除Ollama自动配置
                OllamaChatAutoConfiguration.class,
                OllamaEmbeddingAutoConfiguration.class,
                //移除Cassandra相关配置
                CassandraVectorStoreAutoConfiguration.class,
                //移除Redis相关配置
                RedisVectorStoreAutoConfiguration.class
        }
)
public class MyAiAgentApplication {

    public static void main(String[] args) {
        SpringApplication.run(MyAiAgentApplication.class, args);
    }

}
